import numpy as np

gen_train_indices = False
if gen_train_indices:
    indices = np.array(range(0, 100, 2))
    np.save('train_indices.npy', indices)
    indices = np.load('train_indices.npy')
    print(indices)
